/*
 * Copyright (c) 1999 - 2005 NetGroup, Politecnico di Torino (Italy)
 * Copyright (c) 2005 - 2006 CACE Technologies, Davis (California)
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the Politecnico di Torino, CACE Technologies 
 * nor the names of its contributors may be used to endorse or promote 
 * products derived from this software without specific prior written 
 * permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 */

#ifdef _MSC_VER
/*
 * we do not want the warnings about the old deprecated and unsecure CRT functions
 * since these examples can be compiled under *nix as well
 */
#define _CRT_SECURE_NO_WARNINGS
#endif

#include "pcap.h"
#include <time.h>


int flag = -1;
   //struct tm* tmptime;
int sendlen = 0;
int recivelen = 0;
int ip[4] = { 192,168,159,1 };
/* 6 bytes MAC address */
typedef struct mac_header
{
	u_char srcaddr[6];
	u_char destaddr[6];
	u_char type[2];
}mac_address;

/* 4 bytes IP address */
typedef struct ip_address
{
	u_char byte1;
	u_char byte2;
	u_char byte3;
	u_char byte4;
}ip_address;

/* IPv4 header */
typedef struct ip_header
{
	u_char	ver_ihl;		// Version (4 bits) + Internet header length (4 bits)
	u_char	tos;			// Type of service 
	u_short tlen;			// Total length 
	u_short identification; // Identification
	u_short flags_fo;		// Flags (3 bits) + Fragment offset (13 bits)
	u_char	ttl;			// Time to live
	u_char	proto;			// Protocol
	u_short crc;			// Header checksum
	ip_address	saddr;		// Source address
	ip_address	daddr;		// Destination address
	u_int	op_pad;			// Option + Padding
}ip_header;

/* UDP header*/
typedef struct udp_header
{
	u_short sport;			// Source port
	u_short dport;			// Destination port
	u_short len;			// Datagram length
	u_short crc;			// Checksum
}udp_header;

/* prototype of the packet handler */
void packet_handler(u_char *param, const struct pcap_pkthdr *header, const u_char *pkt_data);
int Issame(int a[4], ip_address add);

#define FROM_NIC
int main()
{
	//	tmptime = time(NULL);
	
	pcap_if_t *alldevs;
	pcap_if_t *d;
	int inum;
	int i=0;
	pcap_t *adhandle;
	char errbuf[PCAP_ERRBUF_SIZE];
	u_int netmask;
	char packet_filter[] = "ip and udp";
	struct bpf_program fcode;
	
#ifdef  FROM_NIC
	
	
	/* Retrieve the device list */  //获取列表
	if(pcap_findalldevs(&alldevs, errbuf) == -1)
	{
		fprintf(stderr,"Error in pcap_findalldevs: %s\n", errbuf);
		exit(1);
	}
	
	/* Print the list */    //打印设备列表
	for(d=alldevs; d; d=d->next)
	{
		printf("%d. %s", ++i, d->name);
		if (d->description)
			printf(" (%s)\n", d->description);
		else
			printf(" (No description available)\n");
	}
	
	if(i==0)
	{
		printf("\nNo interfaces found! Make sure WinPcap is installed.\n");
		return -1;
	}
	
	printf("Enter the interface number (1-%d):",i);
	scanf("%d", &inum);
	
	/* Check if the user specified a valid adapter */  //检查用户输入的网卡的编号
	if(inum < 1 || inum > i)
	{
		printf("\nAdapter number out of range.\n");
		
		/* Free the device list */
		pcap_freealldevs(alldevs);
		return -1;
	}
	
	/* Jump to the selected adapter */  //打开适配器
	for(d=alldevs, i=0; i< inum-1 ;d=d->next, i++);
	
	/* Open the adapter */
	if ((adhandle= pcap_open_live(d->name,	// name of the device
		65536,			// portion of the packet to capture. 
		// 65536 grants that the whole packet will be captured on all the MACs.
		1,				// promiscuous mode (nonzero means promiscuous)
		1000,			// read timeout
		errbuf			// error buffer
		)) == NULL)
	{
		fprintf(stderr,"\nUnable to open the adapter. %s is not supported by WinPcap\n");
		/* Free the device list */
		pcap_freealldevs(alldevs);
		return -1;
	}
	
	/* Check the link layer. We support only Ethernet for simplicity. */  //检查数据链路层
	if(pcap_datalink(adhandle) != DLT_EN10MB)
	{
		fprintf(stderr,"\nThis program works only on Ethernet networks.\n");
		/* Free the device list */
		pcap_freealldevs(alldevs);
		return -1;
	}
	
	if(d->addresses != NULL)
		/* Retrieve the mask of the first address of the interface */         //获取网卡掩码
		netmask=((struct sockaddr_in *)(d->addresses->netmask))->sin_addr.S_un.S_addr;
	else
		/* If the interface is without addresses we suppose to be in a C class network */
		netmask=0xffffff; 
	
	
	//compile the filter          //编译过滤器
	if (pcap_compile(adhandle, &fcode, packet_filter, 1, netmask) <0 )
	{
		fprintf(stderr,"\nUnable to compile the packet filter. Check the syntax.\n");
		/* Free the device list */
		pcap_freealldevs(alldevs);
		return -1;
	}
	
	//set the filter
	if (pcap_setfilter(adhandle, &fcode)<0)
	{
		fprintf(stderr,"\nError setting the filter.\n");
		/* Free the device list */
		pcap_freealldevs(alldevs);
		return -1;
	}
	
	printf("\nlistening on %s...\n", d->description);
	
	/* At this point, we don't need any more the device list. Free it */
	pcap_freealldevs(alldevs);
	
	/* start the capture */
	pcap_loop(adhandle, 0, packet_handler, NULL);
#else
	/* Open the capture file */
	if ((adhandle = pcap_open_offline("C:\\Users\\13763\\Desktop\\dns.pcap",			// name of the device
		errbuf					// error buffer
		)) == NULL)
	{
		fprintf(stderr, "\nUnable to open the file %s.\n");
		return -1;
	}
	
	/* read and dispatch packets until EOF is reached */
	pcap_loop(adhandle, 0, packet_handler, NULL);
	
	pcap_close(adhandle);
	
#endif
	return 0;
}

/* Callback function invoked by libpcap for every incoming packet */
void packet_handler(u_char *param, const struct pcap_pkthdr *header, const u_char *pkt_data)
{
	struct tm* ltime;
	char timestr[16];
	ip_header* ih;
	mac_address* mh;
	udp_header* uh;
	u_int ip_len;
	u_short sport, dport;
	time_t local_tv_sec;
	
	/*
	 * unused parameter
	 */
	//	(VOID)(param);
	
	/* convert the timestamp to readable format */
	//local_tv_sec = header->ts.tv_sec;
	//ltime = localtime(&local_tv_sec);
	//printf("flag:   %d\n", flag);
	//if (!flag || difftime(ltime,tmptime) > 1)
	//{
	////	printf("%s\n", mktime(&tmptime));
	//	flag = 1;
	//	tmptime = time(NULL);
	//	printf("\n总长度：  %d\n", alllen);
	//	alllen = 0;
	//}
	
	//strftime(timestr, sizeof timestr, "%Y-%m-%d", ltime);
	
	/* print timestamp and length of the packet */
	//printf("%s   len:%d ", timestr, header->len);
	//alllen += header->len;
	
	/* retireve the position of the mac header */
	mh = (mac_header*)pkt_data;
	
	/* retireve the position of the ip header */
	ih = (ip_header*)(pkt_data + 14); //length of ethernet header
	FILE* file = fopen("C:\\Users\\13763\\Desktop\\output.txt", "a");
	time_t tt = time(NULL);
	struct tm* t = localtime(&tt);
	
	if (flag == 59)  //下一个小时
	{
		flag = -1;
	}
	if (t->tm_min > flag)
	{
		printf( "%d-%d-%d %d:%.2d:%.2d,", t->tm_year + 1900, t->tm_mon + 1, t->tm_mday, t->tm_hour, t->tm_min, t->tm_sec);
		printf("收到的长度： %d\t发送的长度： %d\n", recivelen,sendlen);
		sendlen = 0;
		recivelen = 0;
		flag = t->tm_min;
	}
	
	fprintf_s(file, "%d-%d-%d %d:%.2d:%.2d,", t->tm_year + 1900, t->tm_mon + 1, t->tm_mday, t->tm_hour, t->tm_min, t->tm_sec);
	for (int i = 0; i < 5; i++) {
		fprintf_s(file, "%02X-", mh->srcaddr[i]);//源MAC地址
	}
	fprintf_s(file, "%02X,", mh->srcaddr[5]);
	
	fprintf_s(file, "%d.", ih->saddr.byte1);//源IP地址
	fprintf_s(file, "%d.", ih->saddr.byte2);
	fprintf_s(file, "%d.", ih->saddr.byte3);
	fprintf_s(file, "%d,", ih->saddr.byte4);
	for (int i = 0; i < 5; i++) {
		fprintf_s(file, "%02X-", mh->destaddr[i]);//目的MAC地址
	}
	fprintf_s(file, "%02X,", mh->destaddr[5]);
	fprintf_s(file, "%d.", ih->daddr.byte1);
	fprintf_s(file, "%d.", ih->daddr.byte2);
	fprintf_s(file, "%d.", ih->daddr.byte3);
	fprintf_s(file, "%d,", ih->daddr.byte4);
	fprintf_s(file, "%d", ntohs(ih->tlen));
	fprintf_s(file, "\n");
	fclose(file);
	
	if (Issame(ip, ih->saddr)) sendlen = sendlen + ih->tlen;
	if (Issame(ip, ih->daddr)) recivelen = recivelen + ih->tlen;
	
	//printf("    %d     ", ih->tlen);
	/*
	 * unused parameter
	 */
	/* retireve the position of the udp header */
	//ip_len = (ih->ver_ihl & 0xf) * 4;
	//uh = (udp_header *) ((u_char*)ih + ip_len);
	/* convert from network byte order to host byte order */
	//sport = ntohs( uh->sport );
	//dport = ntohs( uh->dport );
	
	//要写进文件里去
	/* print ip addresses and udp ports */
	//printf("%d.%d.%d.%d -> %d.%d.%d.%d\t",
	//	ih->saddr.byte1,
	//	ih->saddr.byte2,
	//	ih->saddr.byte3,
	//	ih->saddr.byte4,
	//	ih->daddr.byte1,
	//	ih->daddr.byte2,
	//	ih->daddr.byte3,
	//	ih->daddr.byte4);
	
	//printf("%d.%d.%d.%d.%d.%d-> %d.%d.%d.%d.%d.%d\n",
	//	mh->srcaddr[0],
	//	mh->srcaddr[1],
	//	mh->srcaddr[2],
	//	mh->srcaddr[3],
	//	mh->srcaddr[4],
	//	mh->srcaddr[5],
	//	mh->destaddr[0],
	//	mh->destaddr[1],
	//	mh->destaddr[2],
	//	mh->destaddr[3],
	//	mh->destaddr[4],
	//	mh->destaddr[5]);
	
	/*int res;
	while ((res = pcap_next_ex(param, &header, &pkt_data)) >= 0) {
		if (res == 0) continue;

		struct ethhdr* eth = (struct ethhdr*)(pkt_data);
		unsigned char* src_mac = eth->h_source;

		printf("MAC Address: %02x:%02x:%02x:%02x:%02x:%02x\n",
			src_mac[0], src_mac[1], src_mac[2], src_mac[3], src_mac[4], src_mac[5]);
	}

	pcap_close(param);*/
	return 0;
}

int Issame(int a[4], ip_address add)
{
	if (a[0] != add.byte1) return 0;
	if (a[1] != add.byte2) return 0;
	if (a[2] != add.byte3) return 0;
	if (a[3] != add.byte4) return 0;
	return 1;
}

